import torch

# 检查 CUDA 是否可用
print("CUDA 可用:", torch.cuda.is_available())

# 查看 CUDA 版本
if torch.cuda.is_available():
    print("CUDA 版本:", torch.version.cuda)
    print("GPU 数量:", torch.cuda.device_count())
    print("当前 GPU 索引:", torch.cuda.current_device())
    print("当前 GPU 名称:", torch.cuda.get_device_name(torch.cuda.current_device()))
else:
    print("CUDA 不可用，请检查安装或驱动。")